package com.ideal.util;


import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.naming.Name;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/**
 * 图片本地化工具类
 *
 * @author ideal-20
 * @date 2022-10-30 14:21
 **/
@Component
@Slf4j
public class GetUrlImgUtil {
//    public static void main(String[] args) throws Exception {
//        toLocal("https://cdn.nlark.com/yuque/0/2022/png/22295078/1663838556230-2fbc3671-ff52-4b11-932d-c8c14457297b.png",
//                "C:\\ideal-20\\develop\\docs\\test\\old\\images");
//    }

    /**
     * 图片本地化工具类
     *
     * @param urlStr 网络图片地址
     * @param writePathCommon 写入路径的通用部分，即当前重构文件所在的路径
     * @param targetImgRelativeDirPath 目标生成文件夹相对路径
     * @return md 中应该替换引用的 相对路径
     * @throws Exception 异常外层处理
     */
    public static String toLocal(String urlStr,  String writePathCommon, String targetImgRelativeDirPath) throws Exception {
        // new一个URL对象
        URL url = new URL(urlStr);
        // 打开链接
        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
        // 设置请求方式为"GET"
        conn.setRequestMethod("GET");
        // 超时响应时间为5秒
        conn.setConnectTimeout(5 * 1000);
        // 通过输入流获取图片数据
        InputStream inStream = conn.getInputStream();
        // 得到图片的二进制数据，以二进制封装得到数据，具有通用性
        byte[] data = readInputStream(inStream);
        // new一个文件对象用来保存图片，默认保存当前工程根目录
        String finalImgPath = targetImgRelativeDirPath + chooseName(urlStr, writePathCommon + targetImgRelativeDirPath);
        File imageFile = new File(writePathCommon + finalImgPath);
        // 创建输出流
        FileOutputStream outStream = new FileOutputStream(imageFile);
        // 写入数据
        outStream.write(data);
        // 关闭输出流
        outStream.close();
        // 返回真正的 img 路径例如 images/1.png
        log.info("toLocal return finalImgPath:{}", finalImgPath);
        return finalImgPath;
    }

    private static String chooseName(String urlStr, String path) {
        // 路径
        File imgDir = new File(path);
        // 路径不存在
        if (!imgDir.exists()) {
            boolean mkdir = imgDir.mkdir();
            log.info("imgDir not exists mkdir:{}", mkdir);
            if (!mkdir) {
                throw new RuntimeException("创建 IMG 文件夹失败");
            }
        }

        File[] result = imgDir.listFiles();
        if (result == null || result.length == 0) {
            String suffix = urlStr.substring(urlStr.lastIndexOf("."));
            return "/" + "1" + suffix;
        }
        List<Integer> fileNames = new ArrayList<>();
        // 循环遍历
        for (File file : result) {
            if (file.isFile()) {
                String fileName = file.getName();
                String fileNameExceptSuffix = fileName.substring(0, fileName.lastIndexOf("."));
                fileNames.add(Integer.parseInt(fileNameExceptSuffix));
            }
        }
        Collections.sort(fileNames);
        // 文件名
        Integer curMaxName = fileNames.get(fileNames.size() - 1);
        // 文件后缀
        String suffix = urlStr.substring(urlStr.lastIndexOf("."));
        return "/" + (curMaxName + 1) + suffix;
    }

    public static byte[] readInputStream(InputStream inStream) throws Exception {
        ByteArrayOutputStream outStream = new ByteArrayOutputStream();
        // 创建一个Buffer字符串
        byte[] buffer = new byte[1024];
        // 每次读取的字符串长度，如果为-1，代表全部读取完毕
        int len = 0;
        // 使用一个输入流从buffer里把数据读取出来
        while ((len = inStream.read(buffer)) != -1) {
            //用输出流往buffer里写入数据，中间参数代表从哪个位置开始读，len代表读取的长度
            outStream.write(buffer, 0, len);
        }
        // 关闭输入流
        inStream.close();
        // 把outStream里的数据写入内存
        return outStream.toByteArray();
    }

}